import torch
rand_num=torch.rand(3,4)
rand_num=rand_num.unsqueeze(0)
print(rand_num.size())